From f9b84b2fa9a75c0ffc894f50cfeb43e5733b90b7 Mon Sep 17 00:00:00 2001 From: "Karl O. Pinc" Date: Sun, 29 Sep 2024 17:03:04 -0500 Subject: [PATCH] Make the Include SQL (in the download) checkbox work --- src/pgwui_sql/views/sql.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/src/pgwui_sql/views/sql.py b/src/pgwui_sql/views/sql.py index b3acd11..cdc2fe2 100644 --- a/src/pgwui_sql/views/sql.py +++ b/src/pgwui_sql/views/sql.py @@ -282,6 +282,8 @@ class SQLResultsHandler(pgwui_core.core.SessionDBHandler): if downloading: self.tfile = tempfile.TemporaryFile(mode='w+t', newline='') writer = self.make_csv_writer() + if self.uf['include_sql']: + writer.writerow((self.uf['sql'].rstrip(),)) for sql_result in sql_results: writer.writerow((sql_result.statusmessage.data,)) writer.writerow(sql_result.heading.data) -- 2.34.1